%load_ext pretty_jupyter

import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns
import numpy as np
from pretty_jupyter.helpers import matplotlib_fig_to_markdown

sns.set_theme()

%%jmd

# Algorithm
[//]: # (-.- .tabset .tabset-pills .tabset-fade)

Sometimes we want to create tabsets dynamically, meaning we want to generate them based on a code.
Pretty Jupyter supports this by using Jinja Markdown cells.

# create dataframe
data = pd.DataFrame({"a": [1, 2, 3, 4], "b": [4, 3, 2, 1], "c": [1, 4, 2, 3]})

%%jmd

{% for col in data.columns %}

## Tabset for column {{ col }}

{% set ax = sns.lineplot(x=np.arange(data.shape[0]), y=data[col], marker="o") %}
{% set _ = ax.set(title="Plot for column '{}'".format(col), ylabel=col) %}
{{ matplotlib_fig_to_markdown(ax.figure) }}

{% endfor %}